1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
| #include<iostream> #include<opencv2/opencv.hpp> #include"mydft.h"
using namespace std; using namespace cv;
int main(){
Mat img,img_gray,img_out,img_transform; img = imread("/home/v/home.png"); if(img.empty()){ cout<<"could not load image..."<<endl; return -1; } imshow("img",img);
cvtColor(img,img_gray,COLOR_BGR2GRAY); imshow("img_gray",img_gray);
mydft(img_gray,img_out,img_transform); imshow("img_out",img_out);
Mat planes[] = {Mat_<float>(img_out),Mat::zeros(img_out.size(),CV_32F)}; split(img_transform,planes); Mat img_transform_real = planes[0]; Mat img_transform_imag = planes[1];
int core_x = img_transform_real.rows/2; int core_y = img_transform_real.cols/2; int r = 20; for(int i=0;i<img_transform_real.rows;i++){ for(int j=0;j<img_transform_real.cols;j++){ if(((i-core_x)*(i-core_x)+(j-core_y)*(j-core_y))>r*r){ img_transform_real.at<float>(i,j) = 0; img_transform_imag.at<float>(i,j) = 0; } } }
planes[0] = img_transform_real; planes[1] = img_transform_imag; Mat img_transform_ilpf; merge(planes,2,img_transform_ilpf);
Mat iDft[] = {Mat_<float>(img_out),Mat::zeros(img_out.size(),CV_32F)}; idft(img_transform_ilpf,img_transform_ilpf); split(img_transform_ilpf,iDft); magnitude(iDft[0],iDft[1],img_out); normalize(img_out,img_out,0,1,NORM_MINMAX); imshow("idft",iDft[0]);
waitKey(0); return 0; }
|